package com.yq.dataStructures.tree.binarySortTree;

/**
 * @ClassName BinarySortTreeDemo
 * @Description TODO
 * @Author yaoqiang
 * @Date 2021/11/18 10:55
 * @Version 1.0
 */
public class BinarySortTreeDemo {
    public static void main(String[] args) {
        int[] arr ={7,3,10,12,5,1,9,2};
        BinarySortTree tree=new BinarySortTree();
        //循环将节点添加到二叉排序树中
        for (int i=0;i<arr.length;i++){
            tree.add(new Node(arr[i]));
        }

        tree.delNode(10);
        tree.infixOrder();
    }
}
class BinarySortTree{
    private Node root;
    public void add (Node node){
        //树为空，这个节点直接赋值，否则去调用node的add来判断实现左右子树
        if(root ==null){
            root=node;
        }else {
            root.add(node);
        }
    }
    public void infixOrder(){
        //省略了判断root空
        root.infixOrder();
    }


    public void delNode(int value) {
        if(root !=null){
            //先找要删除的节点
            Node targetNode =search(value);
            if(targetNode==null){return;}
            //如果只有一个节点
            if (root.left==null&& root.right==null){
                root=null;
            }
            //去找target的父节点
            Node parent=searchParent(value);
            //如果要删除的节点是叶子节点,没有子节点
            if(targetNode.left==null && targetNode.right==null){
                //判断是父节点的左节点还是有节点
                if(parent.left!=null && parent.left.value==value){
                    parent.left=null;
                }
                if (parent.right!=null && parent.right.value==value){
                    parent.right=null;
                }
            }else if(targetNode.left!=null && targetNode.right!=null){
                //删除有两颗子树的节点
                int minVal=delRightTreeMin(targetNode.right);
                targetNode.value=minVal;
            }else {
                //删除只有一个子树的节点
                //删除有左子树的节点
                if (targetNode.left!=null){
                    if(parent!=null){
                        //如果target是parent的左子节点
                        if(parent.left.value==value){
                            parent.left=targetNode.left;
                        }else {
                            //parent的右子节点
                            parent.right=targetNode.left;
                        }
                    }else {
                        root=targetNode.left;
                    }
                }else {
                    if (parent!=null){
                        if (parent.right.value==value){
                            parent.left=targetNode.right;
                        }else {
                            parent.right=targetNode.right;
                        }
                    }else {
                        root=targetNode.right;
                    }
                }
            }


        }else {
            return;
        }
    }

    private int delRightTreeMin(Node node) {
        Node target =node;
        while (target.left!=null){
            target=target.left;
        }
        //删除最小及诶单
        delNode(target.value);
        return target.value;
    }

    private Node searchParent(int value) {
        return root.searchParent(value);
    }

    private Node search(int value) {
        return root.search(value);
    }
}
class  Node{
    int value;
    Node left;
    Node right;

    public Node(int value) {
        this.value = value;
    }

    @Override
    public String toString() {
        return "Node{" +
                "value=" + value +
                '}';
    }

    /**
     * 查找要删除的点
     * @param value
     * @return
     */
    public Node search(int value){
        if(value==this.value){
            return this;
        }else  if(this.value>value){
            //比当前节点小，查左子树
            if(this.left==null){
                return  null;
            }
            return this.left.search(value);
        }else {
            if (this.right==null){
                return null;
            }
            return this.right.search(value);
        }
    }

    /**
     * 查找要删除节点的值，如果没有返回null,有就返回要删除节点的父节点
     * @param value
     * @return
     */
    public  Node searchParent(int value){
        //如果当前节点的就是要删除节点的父节点，就返回
        if ((this.left != null && this.left.value == value) ||(this.right!=null && this.right.value==value)) {
            return this;
        }
        //如果查找的值小于当前节点的值并且左子节点不为空
        if (value<this.value && this.left !=null){
            return this.left.searchParent(value);
        }else if(this.right!=null && this.value<value){
            return this.right.searchParent(value);
        }else {
            return  null;
        }
    }

    public void add(Node node){
       //递归的形式添加节点，需要满足二叉树的要求
       if(node ==null){
           return;
       }
       //判断传入节点的值，和当前子树根节点的值的关系
        if(node.value<this.value){
            if(this.left==null){
                this.left=node;
            }else {
                this.left.add(node);
            }
        }else {
            if(this.right==null){
                this.right=node;
            }else {
                this.right.add(node);
            }
        }
    }

    public void infixOrder(){
        if(this.left!=null){
            this.left.infixOrder();
        }
        System.out.println(this);
        if(this.right!=null){
            this.right.infixOrder();
        }
    }


}